#pragma once

#include <iostream>
#include <string>
#include <vector>
#include <ctemplate/template.h>

#include "oj_model.hpp"

namespace ns_view
{
    using namespace ns_model;
    // class Question
    // {
    // public:
    //     std::string number; // 题号
    //     std::string title;  // 题目
    //     std::string star;   // 难度
    //     int cpu_limit;      // 时间限制
    //     int mem_limit;      // 空间限制
    //     std::string desc;   // 题目描述
    //     std::string header; // 用户写入代码
    //     std::string tail;   // 测试用例
    // };

    const std::string template_path = "./template_html/";

    class View
    {
    public:
        View() {}
        ~View() {}

        void AllExpandHtml(const std::vector<Question> &questions, std::string *html)
        {
            // 题目编号 题目标题 题目难度

            // 形成路径
            std::string src_html = template_path + "all_questions.html";
            // 形成数据字典
            ctemplate::TemplateDictionary root("all_questions");
            for (const auto &q : questions)
            {
                ctemplate::TemplateDictionary *sub = root.AddSectionDictionary("question_list");
                sub->SetValue("number", q.number);
                sub->SetValue("title", q.title);
                sub->SetValue("star", q.star);
            }
            // 获取被渲染的html
            ctemplate::Template *tpl = ctemplate::Template::GetTemplate(src_html, ctemplate::DO_NOT_STRIP);
            // 开始完成渲染功能

            tpl->Expand(html, &root);
        }

        void OneExpandHtml(const Question &q, std::string *html)
        {
            // 形成路径
            std::string src_html = template_path + "one_question.html";
            // 形成数据字典
            ctemplate::TemplateDictionary root("one_question");
            root.SetValue("number", q.number);
            root.SetValue("title", q.title);
            root.SetValue("star", q.star);
            root.SetValue("desc", q.desc);
            root.SetValue("pre_code", q.header);
            // 获取被渲染的html
            ctemplate::Template *tpl = ctemplate::Template::GetTemplate(src_html, ctemplate::DO_NOT_STRIP);
            // 开始完成渲染功能
            tpl->Expand(html, &root);
        }
    };
}